import numpy
import numpy as np

if __name__ == '__main__':
    data = numpy.array([[8, 18, 0.],
                        [8, 15, 0.],
                        [8, 18, 1],
                        [4, 32, 0, ],
                        [4, 28, 1],
                        [4, 31, 0, ]])

    print(data[:, 2] == 1)
    print(data[data[:, 2] == 1, :2])
